package com.wanmait.bootdemo.util;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.io.Serializable;
import java.util.List;

/*
* @Description: 自定义响应数据结构
*                 200：表示成功
*                 500：表示错误，错误信息在msg字段中
*                 501：bean验证错误，不管多少个错误都以map形式返回
*                 502：拦截器拦截到用户token出错
*                 503：异常抛出信息
*/
public class JSONResult implements Serializable {
    // 定义jackson对象
    private static final ObjectMapper MAPPER = new ObjectMapper();
    private Integer status;
    private String msg;
    private Object data;
    public JSONResult()
    {

    }
    public JSONResult(Object data)
    {
        this.msg="OK";
        this.status=200;
        this.data=data;
    }
    public JSONResult(Integer status, String msg, Object data)
    {
        this.msg=msg;
        this.status=status;
        this.data=data;
    }

    public static JSONResult build(Integer status,String msg,Object data)
    {
        return new JSONResult(status,msg,data);
    }
    public static JSONResult ok()
    {
        return new JSONResult();
    }

    public static JSONResult ok(Object data)
    {
        return new JSONResult(data);
    }
    public static JSONResult errorMsg(String msg) {
        return new JSONResult(500, msg, null);
    }

    public static JSONResult errorMap(Object data) {
        return new JSONResult(501, "error", data);
    }

    public static JSONResult errorTokenMsg(String msg) {
        return new JSONResult(502, msg, null);
    }

    public static JSONResult errorException(String msg) {
        return new JSONResult(503, msg, null);
    }

    public Boolean isOK() {
        return this.status == 200;
    }

    //将json结果集转化为JSONResult对象  需要转换的对象是一个类
    public static JSONResult formatToPojo(String jsonData, Class<?> clazz) {
        try {
            if (clazz == null) {
                return MAPPER.readValue(jsonData, JSONResult.class);
            }
            JsonNode jsonNode = MAPPER.readTree(jsonData);
            JsonNode data = jsonNode.get("data");
            Object obj = null;
            if (clazz != null) {
                if (data.isObject()) {
                    obj = MAPPER.readValue(data.traverse(), clazz);
                } else if (data.isTextual()) {
                    obj = MAPPER.readValue(data.asText(), clazz);
                }
            }
            return build(jsonNode.get("status").intValue(), jsonNode.get("msg").asText(), obj);
        } catch (Exception e) {
            return null;
        }
    }

    // json字符串转为JSONResult对象 如果data为null
    public static JSONResult format(String json) {
        try {
            return MAPPER.readValue(json, JSONResult.class);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    //转换list集合为JSONResult对象
    public static JSONResult formatToList(String jsonData, Class<?> clazz) {
        try {
            JsonNode jsonNode = MAPPER.readTree(jsonData);
            JsonNode data = jsonNode.get("data");
            Object obj = null;
            if (data.isArray() && data.size() > 0) {
                obj = MAPPER.readValue(data.traverse(),
                        MAPPER.getTypeFactory().constructCollectionType(List.class, clazz));
            }
            return build(jsonNode.get("status").intValue(), jsonNode.get("msg").asText(), obj);
        } catch (Exception e) {
            return null;
        }
    }
    public Integer getStatus() {
        return status;
    }

    public void setStatus(Integer status) {
        this.status = status;
    }

    public String getMsg() {
        return msg;
    }

    public void setMsg(String msg) {
        this.msg = msg;
    }

    public Object getData() {
        return data;
    }

    public void setData(Object data) {
        this.data = data;
    }

}
